import tensorflow as tf

tf_ones = tf.ones(shape=[4, 4])
tf_reshape = tf.reshape(tf_ones, shape=[-1, 2])
print(tf_reshape.shape)
